package com.ego.common.util;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ego.common.vo.PageVo;
import org.springframework.util.CollectionUtils;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @program: ego
 * @description: 分页工具类
 * @author: KamTang
 * @create: 2022-11-11 10:00
 **/
public class PageUtil<T> implements Serializable {

    public static <F, T> PageVo<T> toPage(Page<F> page, Function<F, T> function) {
        PageVo<T> result = PageVo.<T>builder().pageNum((int) page.getCurrent()).pageSize((int) page.getSize())
                .totalPage(page.getPages()).total(page.getTotal()).build();
        result.setData(CollectionUtils.isEmpty(page.getRecords()) ?
                new ArrayList<>() : page.getRecords().stream().map(function).collect(Collectors.<T>toList()));
        return result;
    }

    public static <T> PageVo<T> toPage(Page<T> page) {
        PageVo<T> result = PageVo.<T>builder().pageNum((int) page.getCurrent()).pageSize((int) page.getSize())
                .totalPage(page.getPages()).total(page.getTotal()).build();
        result.setData(CollectionUtils.isEmpty(page.getRecords()) ?
                new ArrayList<>() : new ArrayList<>(page.getRecords()));
        return result;
    }

    public static <T> PageVo<T> toPage(Page<?> page, List<T> list) {
        PageVo<T> result = PageVo.<T>builder().pageNum((int) page.getCurrent()).pageSize((int) page.getSize())
                .totalPage(page.getPages()).total(page.getTotal()).build();
        result.setData(CollectionUtils.isEmpty(list) ?
                new ArrayList<>() : list);
        return result;
    }
}
